import logging.config
import yaml
from pathlib import Path

from locomo import evaluate_qa, evaluate_on_one_conversation


def set_logging_config():
    LOGGING_FILE = "logging_config.yaml"
    with Path(LOGGING_FILE).open("r") as f:
        config = yaml.safe_load(f)
    logging.config.dictConfig(config)


if __name__ == "__main__":
    MEMORY_STORAGE_PATH = "./memory"
    DATA_DIR = "./locomo/data"
    set_logging_config()
    results = evaluate_on_one_conversation(MEMORY_STORAGE_PATH, DATA_DIR)
    print(results)
